#include "ffmpeg_impl.h"
#include "string.h"
#include <dlfcn.h>
#include <QDebug>
FFMPEG_Impl::FFMPEG_Impl()
{
    pFrame_         = 0;
    codec_          = 0;
    videoCodec      = 0;

    pFrameRGB24     = 0;
    pFrameYUV       = 0;
    img_convert_ctx = 0;

    bufferYUV = 0;
    bufferRGB = 0;
}
FFMPEG_Impl::~FFMPEG_Impl()
{
    colse();
}

void FFMPEG_Impl::Init(int width  ,int height )
{
    colse();

    avcodec_register_all();

    pFrame_ = NULL;

    /* find the video encoder */
    videoCodec = avcodec_find_decoder(AV_CODEC_ID_H264);

    if (!videoCodec)
    {
        qDebug(QString("p_avcodec_find_decoder执行失败:").append(QString::number((long)videoCodec)).toLocal8Bit().data());
        return  ;
    }

    codec_ = avcodec_alloc_context3(videoCodec);
    if (!codec_)
    {
            fprintf(stderr, "Could not allocate video codec context\n");
            exit(1);
     }

    //初始化参数，下面的参数应该由具体的业务决定
//    codec_->time_base.num = 1;
//    codec_->frame_number = 1; //每包一个视频帧
//    codec_->codec_type = AVMEDIA_TYPE_VIDEO;
//    codec_->bit_rate = 0;
//    codec_->time_base.den = 25;//帧率
//    codec_->width = 176;//视频宽
//    codec_->height = 144;//视频高
    if(videoCodec->capabilities&CODEC_CAP_TRUNCATED)
           codec_->flags|= CODEC_FLAG_TRUNCATED; /* we do not send complete frames */

    if (avcodec_open2(codec_, videoCodec, NULL) < 0)
    {
        qDebug(QString("p_avcodec_open2执行失败").toLocal8Bit().data());
        fprintf(stderr, "Could not open codec\n");
        exit(1);
    }

 //   int width =176;  int height = 144;

    pFrame_ = avcodec_alloc_frame();// Allocate video frame

    img_convert_ctx = sws_getContext(width, height, AV_PIX_FMT_YUV420P, width, height,AV_PIX_FMT_RGB24 , SWS_FAST_BILINEAR, NULL, NULL, NULL);


     pFrameYUV   =avcodec_alloc_frame();
     int numBytesYUV = avpicture_get_size(AV_PIX_FMT_YUV420P, width, height);
     bufferYUV = (uint8_t *)malloc(numBytesYUV * sizeof(uint8_t));
     avpicture_fill((AVPicture *)pFrameYUV, bufferYUV, AV_PIX_FMT_YUV420P,width, height);


     pFrameRGB24 =avcodec_alloc_frame();
     int numBytesRGB = avpicture_get_size(AV_PIX_FMT_RGB24, width , height);
     bufferRGB = (uint8_t *)malloc(numBytesRGB * sizeof(uint8_t));
     avpicture_fill((AVPicture *)pFrameRGB24, bufferRGB, AV_PIX_FMT_RGB24,width , height);
}

bool FFMPEG_Impl::decode_video(int &frameFinished, uint8_t *data, int size,unsigned char *rgb24,int &nLen)
{
    AVPacket packet;
    av_init_packet(&packet);

    packet.data = data;//这里填入一个指向完整H264数据帧的指针
    packet.size = size;//这个填入H264数据帧的大小
    int A=avcodec_decode_video2(codec_, pFrame_, &frameFinished, &packet);
    if(A)
    {
        ;
    }
    //qDebug(QString("1....").toLocal8Bit().data());
    if(frameFinished)//成功解码
    {
        //qDebug(QString("2....").toLocal8Bit().data());
        if(nLen <(codec_->width * codec_->height * 3))
        {
           QString str;
           //str.sprintf("nLen:%d  codec_->width:%d  codec_->height:%d  codec_->width * codec_->height * 3 : ",nLen,codec_->width,codec_->height,codec_->width * codec_->height * 3);
           qDebug(str.toLocal8Bit().data());
            return false;
        }
        //qDebug(QString("5....OK").toLocal8Bit().data());
        sws_scale(img_convert_ctx, (const uint8_t* const*)pFrame_->data, pFrame_->linesize, 0, codec_->height, pFrameRGB24->data, pFrameRGB24->linesize);

        memcpy(rgb24,pFrameRGB24->data[0],pFrameRGB24->linesize[0]*codec_->height);

        return true;
    }else
    {
        //qDebug(QString("3....").toLocal8Bit().data());
        return false;
    }

}

void FFMPEG_Impl::YUV420p_to_RGB24_2(unsigned char *yuv420[3], unsigned char *rgb24, int width, int height)
{

    memcpy(pFrameYUV->data[0],yuv420[0],width *height);
    memcpy(pFrameYUV->data[1],yuv420[1],width *height/4);
    memcpy(pFrameYUV->data[2],yuv420[2],width *height/4);

    sws_scale(img_convert_ctx, (const uint8_t* const*)pFrameYUV->data, pFrameYUV->linesize, 0, height, pFrameRGB24->data, pFrameRGB24->linesize);

    memcpy(rgb24,pFrameRGB24->data[0],pFrameRGB24->linesize[0]*height);
}

void FFMPEG_Impl::colse()
{
    if(pFrame_)
    {
        av_free(pFrame_);

        pFrame_ = 0;
    }

    if(codec_)
    {
        avcodec_close(codec_);
        codec_ = 0;
    }
    if(img_convert_ctx)
    {
        sws_freeContext(img_convert_ctx);
        img_convert_ctx = 0;
    }

    if(pFrameRGB24)
    {
        av_free(pFrameRGB24);
        pFrameRGB24 = 0;

    }
    if(pFrameYUV)
    {
         av_free(pFrameYUV);
         pFrameYUV = 0;
    }

    if(bufferYUV)
    {
        free(bufferYUV);
        bufferYUV = 0;
    }
    if(bufferRGB)
    {
        free(bufferRGB);
        bufferRGB = 0;
    }
}
//bool FFMPEG_Impl::YV12_to_RGB24(unsigned char* pYV12, unsigned char* pRGB24, int iWidth, int iHeight)
//{
// if(!pYV12 || !pRGB24)
//    return false;

// const long nYLen = long(iHeight * iWidth);
// const int nHfWidth = (iWidth>>1);

// if(nYLen < 1 || nHfWidth < 1)
//    return false;

// unsigned char* yData = pYV12;
// unsigned char* vData = &yData[nYLen];
// unsigned char* uData = &vData[nYLen>>2];
// if(!uData || !vData)
//    return false;

// int rgb[3];
// int i, j, m, n, x, y;
// m = -iWidth;
// n = -nHfWidth;
// for(y = 0; y < iHeight; y++)
// {
//    m += iWidth;

//    if(!(y % 2))
//        n += nHfWidth;

//    for(x=0; x < iWidth; x++)
//    {
//        i = m + x;
//        j = n + (x>>1);
//        rgb[2] = int(yData[i] + 1.370705 * (vData[j] - 128)); // r分量值
//        rgb[1] = int(yData[i] - 0.698001 * (uData[j] - 128)  - 0.703125 * (vData[j] - 128)); // g分量值
//        rgb[0] = int(yData[i] + 1.732446 * (uData[j] - 128)); // b分量值
//        j = nYLen - iWidth - m + x;
//        i = (j<<1) + j;
//        for(j=0; j<3; j++)
//        {
//            if(rgb[j]>=0 && rgb[j]<=255)
//                pRGB24[i + j] = rgb[j];
//            else
//                pRGB24[i + j] = (rgb[j] < 0) ? 0 : 255;
//        }
//    }
// }

// return true;
//}
//bool FFMPEG_Impl::YUV2RGB(unsigned char * yuvBuf,uint nWidth,uint nHeight,unsigned char * pRgbBuf,unsigned long *len)
//{
//#define PIXELSIZE nWidth * nHeight

//    const int Table_fv1[256]={ -180, -179, -177, -176, -174, -173, -172, -170, -169, -167, -166, -165, -163, -162, -160, -159, -158, -156, -155, -153, -152, -151, -149, -148, -146, -145, -144, -142, -141, -139, -138, -137, -135, -134, -132, -131, -130, -128, -127, -125, -124, -123, -121, -120, -118, -117, -115, -114, -113, -111, -110, -108, -107, -106, -104, -103, -101, -100, -99, -97, -96, -94, -93, -92, -90, -89, -87, -86, -85, -83, -82, -80, -79, -78, -76, -75, -73, -72, -71, -69, -68, -66, -65, -64, -62, -61, -59, -58, -57, -55, -54, -52, -51, -50, -48, -47, -45, -44, -43, -41, -40, -38, -37, -36, -34, -33, -31, -30, -29, -27, -26, -24, -23, -22, -20, -19, -17, -16, -15, -13, -12, -10, -9, -8, -6, -5, -3, -2, 0, 1, 2, 4, 5, 7, 8, 9, 11, 12, 14, 15, 16, 18, 19, 21, 22, 23, 25, 26, 28, 29, 30, 32, 33, 35, 36, 37, 39, 40, 42, 43, 44, 46, 47, 49, 50, 51, 53, 54, 56, 57, 58, 60, 61, 63, 64, 65, 67, 68, 70, 71, 72, 74, 75, 77, 78, 79, 81, 82, 84, 85, 86, 88, 89, 91, 92, 93, 95, 96, 98, 99, 100, 102, 103, 105, 106, 107, 109, 110, 112, 113, 114, 116, 117, 119, 120, 122, 123, 124, 126, 127, 129, 130, 131, 133, 134, 136, 137, 138, 140, 141, 143, 144, 145, 147, 148, 150, 151, 152, 154, 155, 157, 158, 159, 161, 162, 164, 165, 166, 168, 169, 171, 172, 173, 175, 176, 178 };
//    const int Table_fv2[256]={ -92, -91, -91, -90, -89, -88, -88, -87, -86, -86, -85, -84, -83, -83, -82, -81, -81, -80, -79, -78, -78, -77, -76, -76, -75, -74, -73, -73, -72, -71, -71, -70, -69, -68, -68, -67, -66, -66, -65, -64, -63, -63, -62, -61, -61, -60, -59, -58, -58, -57, -56, -56, -55, -54, -53, -53, -52, -51, -51, -50, -49, -48, -48, -47, -46, -46, -45, -44, -43, -43, -42, -41, -41, -40, -39, -38, -38, -37, -36, -36, -35, -34, -33, -33, -32, -31, -31, -30, -29, -28, -28, -27, -26, -26, -25, -24, -23, -23, -22, -21, -21, -20, -19, -18, -18, -17, -16, -16, -15, -14, -13, -13, -12, -11, -11, -10, -9, -8, -8, -7, -6, -6, -5, -4, -3, -3, -2, -1, 0, 0, 1, 2, 2, 3, 4, 5, 5, 6, 7, 7, 8, 9, 10, 10, 11, 12, 12, 13, 14, 15, 15, 16, 17, 17, 18, 19, 20, 20, 21, 22, 22, 23, 24, 25, 25, 26, 27, 27, 28, 29, 30, 30, 31, 32, 32, 33, 34, 35, 35, 36, 37, 37, 38, 39, 40, 40, 41, 42, 42, 43, 44, 45, 45, 46, 47, 47, 48, 49, 50, 50, 51, 52, 52, 53, 54, 55, 55, 56, 57, 57, 58, 59, 60, 60, 61, 62, 62, 63, 64, 65, 65, 66, 67, 67, 68, 69, 70, 70, 71, 72, 72, 73, 74, 75, 75, 76, 77, 77, 78, 79, 80, 80, 81, 82, 82, 83, 84, 85, 85, 86, 87, 87, 88, 89, 90, 90 };
//    const int Table_fu1[256]={ -44, -44, -44, -43, -43, -43, -42, -42, -42, -41, -41, -41, -40, -40, -40, -39, -39, -39, -38, -38, -38, -37, -37, -37, -36, -36, -36, -35, -35, -35, -34, -34, -33, -33, -33, -32, -32, -32, -31, -31, -31, -30, -30, -30, -29, -29, -29, -28, -28, -28, -27, -27, -27, -26, -26, -26, -25, -25, -25, -24, -24, -24, -23, -23, -22, -22, -22, -21, -21, -21, -20, -20, -20, -19, -19, -19, -18, -18, -18, -17, -17, -17, -16, -16, -16, -15, -15, -15, -14, -14, -14, -13, -13, -13, -12, -12, -11, -11, -11, -10, -10, -10, -9, -9, -9, -8, -8, -8, -7, -7, -7, -6, -6, -6, -5, -5, -5, -4, -4, -4, -3, -3, -3, -2, -2, -2, -1, -1, 0, 0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 6, 7, 7, 7, 8, 8, 8, 9, 9, 9, 10, 10, 11, 11, 11, 12, 12, 12, 13, 13, 13, 14, 14, 14, 15, 15, 15, 16, 16, 16, 17, 17, 17, 18, 18, 18, 19, 19, 19, 20, 20, 20, 21, 21, 22, 22, 22, 23, 23, 23, 24, 24, 24, 25, 25, 25, 26, 26, 26, 27, 27, 27, 28, 28, 28, 29, 29, 29, 30, 30, 30, 31, 31, 31, 32, 32, 33, 33, 33, 34, 34, 34, 35, 35, 35, 36, 36, 36, 37, 37, 37, 38, 38, 38, 39, 39, 39, 40, 40, 40, 41, 41, 41, 42, 42, 42, 43, 43 };
//    const int Table_fu2[256]={ -227, -226, -224, -222, -220, -219, -217, -215, -213, -212, -210, -208, -206, -204, -203, -201, -199, -197, -196, -194, -192, -190, -188, -187, -185, -183, -181, -180, -178, -176, -174, -173, -171, -169, -167, -165, -164, -162, -160, -158, -157, -155, -153, -151, -149, -148, -146, -144, -142, -141, -139, -137, -135, -134, -132, -130, -128, -126, -125, -123, -121, -119, -118, -116, -114, -112, -110, -109, -107, -105, -103, -102, -100, -98, -96, -94, -93, -91, -89, -87, -86, -84, -82, -80, -79, -77, -75, -73, -71, -70, -68, -66, -64, -63, -61, -59, -57, -55, -54, -52, -50, -48, -47, -45, -43, -41, -40, -38, -36, -34, -32, -31, -29, -27, -25, -24, -22, -20, -18, -16, -15, -13, -11, -9, -8, -6, -4, -2, 0, 1, 3, 5, 7, 8, 10, 12, 14, 15, 17, 19, 21, 23, 24, 26, 28, 30, 31, 33, 35, 37, 39, 40, 42, 44, 46, 47, 49, 51, 53, 54, 56, 58, 60, 62, 63, 65, 67, 69, 70, 72, 74, 76, 78, 79, 81, 83, 85, 86, 88, 90, 92, 93, 95, 97, 99, 101, 102, 104, 106, 108, 109, 111, 113, 115, 117, 118, 120, 122, 124, 125, 127, 129, 131, 133, 134, 136, 138, 140, 141, 143, 145, 147, 148, 150, 152, 154, 156, 157, 159, 161, 163, 164, 166, 168, 170, 172, 173, 175, 177, 179, 180, 182, 184, 186, 187, 189, 191, 193, 195, 196, 198, 200, 202, 203, 205, 207, 209, 211, 212, 214, 216, 218, 219, 221, 223, 225 };
//    *len = 3* nWidth * nHeight;
//    if(!yuvBuf || !pRgbBuf)
//        return false;
//    const long nYLen = long(PIXELSIZE);
//    const int nHfWidth = (nWidth>>1);
//    if(nYLen<1 || nHfWidth<1)
//        return false;
//    // Y data
//    unsigned char* yData = yuvBuf;
//    // v data
//    unsigned char* vData = &yData[nYLen];
//    // u data
//    unsigned char* uData = &vData[nYLen>>2];
//    if(!uData || !vData)
//        return false;
//    int rgb[3];
//    int i, j, m, n, x, y, pu, pv, py, rdif, invgdif, bdif;
//    m = -nWidth;
//    n = -nHfWidth;
//    bool addhalf = true;
//    for(y=0; y<nHeight;y++) {
//        m += nWidth;
//        if( addhalf ){
//            n+=nHfWidth;
//            addhalf = false;
//        } else {
//            addhalf = true;
//        }
//        for(x=0; x<nWidth;x++)  {
//            i = m + x;
//            j = n + (x>>1);
//            py = yData[i];
//            // search tables to get rdif invgdif and bidif
//            rdif = Table_fv1[vData[j]];    // fv1
//            invgdif = Table_fu1[uData[j]] + Table_fv2[vData[j]]; // fu1+fv2
//            bdif = Table_fu2[uData[j]]; // fu2
//            rgb[0] = py+rdif;    // R
//            rgb[1] = py-invgdif; // G
//            rgb[2] = py+bdif;    // B
//            j = nYLen - nWidth - m + x;
//            i = (j<<1) + j;
//            // copy this pixel to rgb data
//            for(j=0; j<3; j++)
//            {
//                if(rgb[j]>=0 && rgb[j]<=255){
//                    pRgbBuf[i + j] = rgb[j];
//                }
//                else{
//                    pRgbBuf[i + j] = (rgb[j] < 0)? 0 : 255;
//                }
//            }
//        }
//    }
//    return true;
//}
////平面YUV420转RGB24
//void FFMPEG_Impl::YUV420p_to_RGB24(unsigned char *yuv420[3], unsigned char *rgb24, int width, int height)
//{
////  int begin = GetTickCount();
// int R,G,B,Y,U,V;
// int x,y;
// int nWidth = width>>1; //色度信号宽度
// for (y=0;y<height;y++)
// {
//  for (x=0;x<width;x++)
//  {
//   Y = *(yuv420[0] + y*width + x);
//   U = *(yuv420[1] + ((y>>1)*nWidth) + (x>>1));
//   V = *(yuv420[2] + ((y>>1)*nWidth) + (x>>1));
//   R = Y + 1.402*(V-128);
//   G = Y - 0.34414*(U-128) - 0.71414*(V-128);
//   B = Y + 1.772*(U-128);

//   //防止越界
//   if (R>255)R=255;
//   if (R<0)R=0;
//   if (G>255)G=255;
//   if (G<0)G=0;
//   if (B>255)B=255;
//   if (B<0)B=0;

//   *(rgb24 + ((height-y-1)*width + x)*3) = B;
//   *(rgb24 + ((height-y-1)*width + x)*3 + 1) = G;
//   *(rgb24 + ((height-y-1)*width + x)*3 + 2) = R;
////    *(rgb24 + (y*width + x)*3) = B;
////    *(rgb24 + (y*width + x)*3 + 1) = G;
////    *(rgb24 + (y*width + x)*3 + 2) = R;
//  }
// }
//}


